package cn.bugstack.chatgpt.data.types.enums;

import lombok.AllArgsConstructor;
import lombok.Getter;
import org.checkerframework.checker.units.qual.A;

@Getter
@AllArgsConstructor
public enum OpenAiChannel {
    ChatGLM("ChatGLM"),
    DeepSeek("DeepSeek"),

    ;

    private String code;

    public static OpenAiChannel getChannel(String model) {
        if (model.toLowerCase().contains("seek")) return OpenAiChannel.DeepSeek;
        if (model.toLowerCase().contains("glm")) return OpenAiChannel.ChatGLM;
        return null;
    }
}
